from matplotlib import pyplot as plt

def main():
    # 这意味着我们的画布上有左右两个画板
    fig, axs = plt.subplots(1, 2, figsize=(10, 4)) # figsize可以调整画布大小

    # --- 在左边的画板 (axs[0]) 上画画 ---
    # 你的学习时长
    days = [1, 2, 3, 4, 5]
    my_hours = [1, 2, 3, 2.5, 4]
    axs[0].plot(days, my_hours, label='我的学习')
    # 你朋友的学习时长
    friend_hours = [2, 2.5, 2, 3, 3.5]
    axs[0].plot(days, friend_hours, label='朋友的学习')
    axs[0].set_title("学习时长对比")
    axs[0].set_xlabel("天数")
    axs[0].set_ylabel("小时")
    axs[0].legend() # 显示图例，即 label 的内容

    # --- 在右边的画板 (axs[1]) 上画画 ---
    # 水果销量
    fruits = ['苹果', '香蕉', '橙子']
    sales = [50, 80, 45]
    axs[1].bar(fruits, sales)
    axs[1].set_title("水果销量")
    axs[1].set_xlabel("水果")
    axs[1].set_ylabel("销量 (箱)")

    # 自动调整布局，防止重叠
    plt.tight_layout()

    plt.show()
    plt.savefig("my_plot.png")  # 保存图像到文件


if __name__ == "__main__":
    main()